#include "mystdio.h"
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include <assert.h>

#define mode 0666
_FILE*_fopen(const char*filename,const char *flag)
{
    assert(filename);
    assert(flag);
    int fd=-1;
    if(strcmp(flag,"w")==0)
    {
        fd=open(filename,O_CREAT|O_WRONLY|O_TRUNC,mode);
    }
    else if(strcmp(flag,"a")==0)
    {
        fd=open(filename,O_CREAT|O_WRONLY|O_APPEND,mode);
    }
    else if(strcmp(flag,"r")==0)
    {
        fd=open(filename,O_RDONLY);
    }
    else return NULL;

    if(fd==-1)
    {
        return NULL;
    }

    _FILE*fp=(_FILE*)malloc(sizeof(_FILE));
    if(fp==NULL)return NULL;
    
    fp->fileno=fd;
    fp->flag=FLUSH_ALL;
    fp->out_pos=0;

    return fp;
}
int _fwrite(_FILE*fp,const char*msg,int len)
{
    memcpy(&fp->outbuffer[fp->out_pos],msg,len);
    //没有做异常处理，拷贝时可能空间不够（动态维护缓冲区）
    //也不考虑局部问题：比如需要拷贝8个字节，但是总长5个字节，
    //那就拷贝5个，返回值为5而不是8
    fp->out_pos+=len;

    if(fp->flag&FLUSH_NOW)
    {
        write(fp->fileno,fp->outbuffer,fp->out_pos);
        fp->out_pos=0;
    }
    else if(fp->flag&FLUSH_LINE)
    {
        if(fp->outbuffer[fp->out_pos-1]=='\n')//不考虑其他情况 如：abcd \n def... 只刷新abcd
        {
            write(fp->fileno,fp->outbuffer,fp->out_pos);
            fp->out_pos=0;
        }
    }
    else if(fp->flag&FLUSH_ALL)
    {
        if(fp->out_pos==SIZE)
        {
            write(fp->fileno,fp->outbuffer,fp->out_pos);
            fp->out_pos=0;
        }
    }

    return len;    
}
void _fflush(_FILE*fp)
{
    if(fp->out_pos>0)
    {
        write(fp->fileno,fp->outbuffer,fp->out_pos);
        fp->out_pos=0;
    }
}

void _fclose(_FILE*fp)
{
    if(fp==NULL)return;
    _fflush(fp);
    close(fp->fileno);
    free(fp);
}
